from gen import generate_data
from alg import distribute_cookies

def test():
    cookies, k = generate_data()
    print(f"测试数据: {cookies}, k={k}")
    unfairness = distribute_cookies(cookies, k)
    print(f"最小不公平度: {unfairness}")

if __name__ == "__main__":
    test()